package com.mzj.saas.commons;

import org.apache.poi.hwpf.HWPFDocument;
import org.apache.poi.hwpf.converter.PicturesManager;
import org.apache.poi.hwpf.converter.WordToHtmlConverter;
import org.apache.poi.hwpf.usermodel.PictureType;
import org.apache.poi.xwpf.converter.xhtml.XHTMLConverter;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;

import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.*;

/**
 * word 转换成html
 */
public class WordToHtml {
	private static final Logger log = LoggerFactory.getLogger(WordToHtml.class);

	/**
	 * 2007版本word转换成html
	 * 
	 * @throws IOException
	 */

	public static String Word2007ToHtml(File file) {
		if (!file.exists()) {
			log.warn("Sorry File does not Exists!, file={}", file);
			return null;
		} else {
			ByteArrayOutputStream baos = null;
			try {
				if (file.getName().endsWith(".docx") || file.getName().endsWith(".DOCX")) {
					// 1) 加载word文档生成 XWPFDocument对象
					InputStream in = new FileInputStream(file);
					XWPFDocument document = new XWPFDocument(in);

					// 2) 解析 XHTML配置 (这里设置IURIResolver来设置图片存放的目录)
					/*
					 * File imageFolderFile = new File(filepath); XHTMLOptions
					 * options = XHTMLOptions.create().URIResolver(new
					 * FileURIResolver(imageFolderFile));
					 * options.setExtractor(new
					 * FileImageExtractor(imageFolderFile));
					 * options.setIgnoreStylesIfUnused(false);
					 * options.setFragment(true);
					 */

					// 3) 将 XWPFDocument转换成XHTML
					// OutputStream out = new FileOutputStream(new File(filepath
					// + htmlName));
					// XHTMLConverter.getInstance().convert(document, out,
					// options);

					// 也可以使用字符数组流获取解析的内容
					baos = new ByteArrayOutputStream();
					XHTMLConverter.getInstance().convert(document, baos, null);
					String content = baos.toString();
					return content;
				} else {
					log.warn("Enter only MS Office 2007+ files!, file={}", file);
				}
			} catch (Exception e) {
				log.error(e.getMessage(), e);
			} finally {
				if (baos != null) {
					try {
						baos.close();
					} catch (IOException e) {
						log.error(e.getMessage(), e);
					}
				}
			}
		}
		return null;
	}

	/**
	 * /** 2003版本word转换成html
	 * 
	 * @throws IOException
	 * @throws TransformerException
	 * @throws ParserConfigurationException
	 */
	public void Word2003ToHtml() throws IOException, TransformerException, ParserConfigurationException {
		final String imagepath = "F:/test/image/";// 解析时候如果doc文件中有图片 图片会保存在此路径
		String filepath = "F:/test/";
		String fileName = "123.doc";
		String htmlName = "123.html";
		final String file = filepath + fileName;
		InputStream input = new FileInputStream(new File(file));
		HWPFDocument wordDocument = new HWPFDocument(input);
		WordToHtmlConverter wordToHtmlConverter = new WordToHtmlConverter(
				DocumentBuilderFactory.newInstance().newDocumentBuilder().newDocument());
		// 设置图片存放的位置
		wordToHtmlConverter.setPicturesManager(new PicturesManager() {
			public String savePicture(byte[] content, PictureType pictureType, String suggestedName, float widthInches,
					float heightInches) {
				File imgPath = new File(imagepath);
				if (!imgPath.exists()) {// 图片目录不存在则创建
					imgPath.mkdirs();
				}
				File file = new File(imagepath + suggestedName);
				try {
					OutputStream os = new FileOutputStream(file);
					os.write(content);
					os.close();
				} catch (FileNotFoundException e) {
					e.printStackTrace();
				} catch (IOException e) {
					e.printStackTrace();
				}
				return imagepath + suggestedName;
			}
		});

		// 解析word文档
		wordToHtmlConverter.processDocument(wordDocument);
		Document htmlDocument = wordToHtmlConverter.getDocument();

		File htmlFile = new File(filepath + htmlName);
		OutputStream outStream = new FileOutputStream(htmlFile);

		// 也可以使用字符数组流获取解析的内容
		// ByteArrayOutputStream baos = new ByteArrayOutputStream();
		// OutputStream outStream = new BufferedOutputStream(baos);

		DOMSource domSource = new DOMSource(htmlDocument);
		StreamResult streamResult = new StreamResult(outStream);

		TransformerFactory factory = TransformerFactory.newInstance();
		Transformer serializer = factory.newTransformer();
		serializer.setOutputProperty(OutputKeys.ENCODING, "utf-8");
		serializer.setOutputProperty(OutputKeys.INDENT, "yes");
		serializer.setOutputProperty(OutputKeys.METHOD, "html");

		serializer.transform(domSource, streamResult);

		// 也可以使用字符数组流获取解析的内容
		// String content = baos.toString();
		// System.out.println(content);
		// baos.close();
		outStream.close();
	}
}